PropertyTable.java

/*******************************************************************************
 * DialogueEditor
 * Copyright (C) 2013-2014 Pawel Pastuszak
 * <p>
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * <p>
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * <p>
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 ******************************************************************************/

package pl.kotcrab.jdialogue.editor;

import pl.kotcrab.jdialogue.editor.components.ChoiceComponentChoices;
import pl.kotcrab.jdialogue.editor.gui.CallbackJComboBoxModel;
import pl.kotcrab.jdialogue.editor.gui.CharactersJComboBoxModel;
import pl.kotcrab.jdialogue.editor.gui.ChoiceComponentChoicesEditor;
import pl.kotcrab.jdialogue.editor.gui.LeftNumberEditor;
import pl.kotcrab.jdialogue.editor.project.PCallback;
import pl.kotcrab.jdialogue.editor.project.PCharacter;
import pl.kotcrab.jdialogue.editor.project.Project;

import javax.swing.DefaultCellEditor;
import javax.swing.JCheckBox;
import javax.swing.JComboBox;
import javax.swing.JLabel;
import javax.swing.JTable;
import javax.swing.JTextField;
import javax.swing.table.DefaultTableCellRenderer;
import javax.swing.table.TableCellEditor;
import javax.swing.table.TableCellRenderer;
import javax.swing.table.TableModel;
import java.awt.event.FocusEvent;
import java.awt.event.FocusListener;

/**
 * Table with custom renderer
 * @author Pawel Pastuszak
 */
public class PropertyTable extends JTable {
	private static final long serialVersionUID = 1L;

	private DefaultTableCellRenderer leftRenderer = new DefaultTableCellRenderer();
	private DefaultCellEditor textEditor;
	private Project project;
	private JComboBox<PCharacter> characterCombobox;
	private JComboBox<PCallback> callbackCombobox;

	public PropertyTable (TableModel dm) {
		super(dm);
		leftRenderer.setHorizontalAlignment(JLabel.LEFT);

		final JTextField textField = new JTextField();
		textField.setName("Table.editor");
		textField.addFocusListener(new FocusListener() {
			@Override
			public void focusLost (FocusEvent e) {
			}

			@Override
			public void focusGained (FocusEvent e) {
				textField.selectAll();
			}
		});

		textEditor = new DefaultCellEditor(textField);
	}

	@Override
	public TableCellEditor getCellEditor (int row, int column) {
		Object value = super.getValueAt(row, column);
		if (value instanceof Boolean) {
			DefaultCellEditor editor = (DefaultCellEditor) getDefaultEditor(Boolean.class);
			JCheckBox check = (JCheckBox) (editor.getComponent());
			check.setHorizontalAlignment(JLabel.LEFT);
			return editor;
		}

		if (value instanceof String) {
			return textEditor;
		}

		if (value instanceof PCharacter) {
			characterCombobox.updateUI(); // make sure that combobox upadted itself (without this list will be blank if character list changed)
			return new DefaultCellEditor(characterCombobox) {
				@Override
				public boolean stopCellEditing () {
					if (characterCombobox.getSelectedItem() == null) cancelCellEditing();
					return super.stopCellEditing();
				}
			};
		}

		if (value instanceof PCallback) {
			callbackCombobox.updateUI();
			return new DefaultCellEditor(callbackCombobox) {
				@Override
				public boolean stopCellEditing () {
					if (callbackCombobox.getSelectedItem() == null) cancelCellEditing();
					return super.stopCellEditing();
				}
			};
		}

		// TODO za kazdym razem nowy czy stary moze byc?
		if (value instanceof ChoiceComponentChoices) {
			return new ChoiceComponentChoicesEditor();
		}

		if (value instanceof Integer) {
			return new LeftNumberEditor();
		}
		// no special case
		return super.getCellEditor(row, column);
	}

	@Override
	public TableCellRenderer getCellRenderer (int row, int column) {
		Object value = super.getValueAt(row, column);

		if (value instanceof Boolean) {
			TableCellRenderer ren = getDefaultRenderer(Boolean.class);
			JCheckBox check = (JCheckBox) ren;
			check.setHorizontalAlignment(JLabel.LEFT);
			return ren;
		}

		if (value instanceof Integer || value instanceof String || value instanceof ChoiceComponentChoices || value instanceof PCharacter) {
			return leftRenderer;
		}

		// no special case
		return super.getCellRenderer(row, column);
	}

	public Project getProject () {
		return project;
	}

	public void setProject (Project project) {
		this.project = project;
		characterCombobox = new JComboBox<>(new CharactersJComboBoxModel(project.getCharacters()));
		callbackCombobox = new JComboBox<>(new CallbackJComboBoxModel(project.getCallbacks()));
		characterCombobox.setLightWeightPopupEnabled(false);
		callbackCombobox.setLightWeightPopupEnabled(false);
	}

}